from base.base_page import BasePage
from selenium import webdriver
from selenium.webdriver.common.by import By


class SearchPage(BasePage):
    '''百度的搜索功能，具体的业务写在这里'''

    def __init__(self, driver, url):
        BasePage.__init__(self,driver, url)

    ''' 下面都是验证百度搜索的逻辑'''

    # 1. 进入百度
    def go_baidu(self):
        self.get()

    # 2. 定位到搜索框，并且输入要搜索的内容
    def locate_search_input(self, key):
        self.key = key
        self.send_keys(key, By.ID, "kw")

    # 3. 定位到百度一下按钮，点击
    def click_baidu_btn(self):
        self.click(By.XPATH, '//*[@id="su"]')

    # 4.搜索结果出来之后，验证结果，需要把列表标题所有的元素拿到，验证所有标题取单元测试里面验证
    def get_searchResult(self):
        h3s = self.find_elements(By.CLASS_NAME, "t")
        titles = []
        for i in h3s:
            print(i)
            titles.append(i.text)
        return titles




